#!/usr/bin/env python

# !-*-coding:utf-8 -*-

# !@Time     : 2021/11/2 11:05

# !@Author   : xul

# !@File     : testScatter.py

# 数据集
import matplotlib.pyplot as plt
from Coures_second.planar_utils import load_extra_datasets

noisy_circles, noisy_moons, blobs, gaussian_quantiles, no_structure = load_extra_datasets()

datasets = {"noisy_circles": noisy_circles,
            "noisy_moons": noisy_moons,
            "blobs": blobs,
            "gaussian_quantiles": gaussian_quantiles}

dataset = "noisy_moons"

X, Y = datasets[dataset]
X, Y = X.T, Y.reshape(1, Y.shape[0])

if dataset == "blobs":
    Y = Y % 2

plt.scatter(X[0, :], X[1, :], c=Y, s=40, cmap=plt.cm.Spectral)
plt.show()
